home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / urllib2.pyo (.txt) < prev    next >
Python Compiled Bytecode  |  2005-10-18  |  41KB  |  1,307 lines

  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.4)
  3.  
  4. '''An extensible library for opening URLs using a variety of protocols
  5.  
  6. The simplest way to use this module is to call the urlopen function,
  7. which accepts a string containing a URL or a Request object (described
  8. below).  It opens the URL and returns the results as file-like
  9. object; the returned object has some extra methods described below.
  10.  
  11. The OpenerDirector manages a collection of Handler objects that do
  12. all the actual work.  Each Handler implements a particular protocol or
  13. option.  The OpenerDirector is a composite object that invokes the
  14. Handlers needed to open the requested URL.  For example, the
  15. HTTPHandler performs HTTP GET and POST requests and deals with
  16. non-error returns.  The HTTPRedirectHandler automatically deals with
  17. HTTP 301, 302, 303 and 307 redirect errors, and the HTTPDigestAuthHandler
  18. deals with digest authentication.
  19.  
  20. urlopen(url, data=None) -- basic usage is that same as original
  21. urllib.  pass the url and optionally data to post to an HTTP URL, and
  22. get a file-like object back.  One difference is that you can also pass
  23. a Request instance instead of URL.  Raises a URLError (subclass of
  24. IOError); for HTTP errors, raises an HTTPError, which can also be
  25. treated as a valid response.
  26.  
  27. build_opener -- function that creates a new OpenerDirector instance.
  28. will install the default handlers.  accepts one or more Handlers as
  29. arguments, either instances or Handler classes that it will
  30. instantiate.  if one of the argument is a subclass of the default
  31. handler, the argument will be installed instead of the default.
  32.  
  33. install_opener -- installs a new opener as the default opener.
  34.  
  35. objects of interest:
  36. OpenerDirector --
  37.  
  38. Request -- an object that encapsulates the state of a request.  the
  39. state can be a simple as the URL.  it can also include extra HTTP
  40. headers, e.g. a User-Agent.
  41.  
  42. BaseHandler --
  43.  
  44. exceptions:
  45. URLError-- a subclass of IOError, individual protocols have their own
  46. specific subclass
  47.  
  48. HTTPError-- also a valid HTTP response, so you can treat an HTTP error
  49. as an exceptional event or valid response
  50.  
  51. internals:
  52. BaseHandler and parent
  53. _call_chain conventions
  54.  
  55. Example usage:
  56.  
  57. import urllib2
  58.  
  59. # set up authentication info
  60. authinfo = urllib2.HTTPBasicAuthHandler()
  61. authinfo.add_password(\'realm\', \'host\', \'username\', \'password\')
  62.  
  63. proxy_support = urllib2.ProxyHandler({"http" : "http://ahad-haam:3128"})
  64.  
  65. # build a new opener that adds authentication and caching FTP handlers
  66. opener = urllib2.build_opener(proxy_support, authinfo, urllib2.CacheFTPHandler)
  67.  
  68. # install it
  69. urllib2.install_opener(opener)
  70.  
  71. f = urllib2.urlopen(\'http://www.python.org/\')
  72.  
  73.  
  74. '''
  75. import base64
  76. import ftplib
  77. import gopherlib
  78. import httplib
  79. import inspect
  80. import md5
  81. import mimetypes
  82. import mimetools
  83. import os
  84. import posixpath
  85. import random
  86. import re
  87. import sha
  88. import socket
  89. import sys
  90. import time
  91. import urlparse
  92. import bisect
  93. import cookielib
  94.  
  95. try:
  96.     from cStringIO import StringIO
  97. except ImportError:
  98.     from StringIO import StringIO
  99.  
  100. from urllib import unwrap, unquote, splittype, splithost, addinfourl, splitport, splitgophertype, splitquery, splitattr, ftpwrapper, noheaders, splituser, splitpasswd, splitvalue
  101. from urllib import localhost, url2pathname, getproxies
  102. __version__ = '2.4'
  103. _opener = None
  104.  
  105. def urlopen(url, data = None):
  106.     global _opener
  107.     if _opener is None:
  108.         _opener = build_opener()
  109.     
  110.     return _opener.open(url, data)
  111.  
  112.  
  113. def install_opener(opener):
  114.     global _opener
  115.     _opener = opener
  116.  
  117.  
  118. class URLError(IOError):
  119.     
  120.     def __init__(self, reason):
  121.         self.args = (reason,)
  122.         self.reason = reason
  123.  
  124.     
  125.     def __str__(self):
  126.         return '<urlopen error %s>' % self.reason
  127.  
  128.  
  129.  
  130. class HTTPError(URLError, addinfourl):
  131.     '''Raised when HTTP error occurs, but also acts like non-error return'''
  132.     __super_init = addinfourl.__init__
  133.     
  134.     def __init__(self, url, code, msg, hdrs, fp):
  135.         self.code = code
  136.         self.msg = msg
  137.         self.hdrs = hdrs
  138.         self.fp = fp
  139.         self.filename = url
  140.         if fp is not None:
  141.             self._HTTPError__super_init(fp, hdrs, url)
  142.         
  143.  
  144.     
  145.     def __str__(self):
  146.         return 'HTTP Error %s: %s' % (self.code, self.msg)
  147.  
  148.  
  149.  
  150. class GopherError(URLError):
  151.     pass
  152.  
  153.  
  154. class Request:
  155.     
  156.     def __init__(self, url, data = None, headers = { }, origin_req_host = None, unverifiable = False):
  157.         self._Request__original = unwrap(url)
  158.         self.type = None
  159.         self.host = None
  160.         self.port = None
  161.         self.data = data
  162.         self.headers = { }
  163.         for key, value in headers.items():
  164.             self.add_header(key, value)
  165.         
  166.         self.unredirected_hdrs = { }
  167.         if origin_req_host is None:
  168.             origin_req_host = cookielib.request_host(self)
  169.         
  170.         self.origin_req_host = origin_req_host
  171.         self.unverifiable = unverifiable
  172.  
  173.     
  174.     def __getattr__(self, attr):
  175.         if attr[:12] == '_Request__r_':
  176.             name = attr[12:]
  177.             if hasattr(Request, 'get_' + name):
  178.                 getattr(self, 'get_' + name)()
  179.                 return getattr(self, attr)
  180.             
  181.         
  182.         raise AttributeError, attr
  183.  
  184.     
  185.     def get_method(self):
  186.         if self.has_data():
  187.             return 'POST'
  188.         else:
  189.             return 'GET'
  190.  
  191.     
  192.     def add_data(self, data):
  193.         self.data = data
  194.  
  195.     
  196.     def has_data(self):
  197.         return self.data is not None
  198.  
  199.     
  200.     def get_data(self):
  201.         return self.data
  202.  
  203.     
  204.     def get_full_url(self):
  205.         return self._Request__original
  206.  
  207.     
  208.     def get_type(self):
  209.         if self.type is None:
  210.             (self.type, self._Request__r_type) = splittype(self._Request__original)
  211.             if self.type is None:
  212.                 raise ValueError, 'unknown url type: %s' % self._Request__original
  213.             
  214.         
  215.         return self.type
  216.  
  217.     
  218.     def get_host(self):
  219.         if self.host is None:
  220.             (self.host, self._Request__r_host) = splithost(self._Request__r_type)
  221.             if self.host:
  222.                 self.host = unquote(self.host)
  223.             
  224.         
  225.         return self.host
  226.  
  227.     
  228.     def get_selector(self):
  229.         return self._Request__r_host
  230.  
  231.     
  232.     def set_proxy(self, host, type):
  233.         self.host = host
  234.         self.type = type
  235.         self._Request__r_host = self._Request__original
  236.  
  237.     
  238.     def get_origin_req_host(self):
  239.         return self.origin_req_host
  240.  
  241.     
  242.     def is_unverifiable(self):
  243.         return self.unverifiable
  244.  
  245.     
  246.     def add_header(self, key, val):
  247.         self.headers[key.capitalize()] = val
  248.  
  249.     
  250.     def add_unredirected_header(self, key, val):
  251.         self.unredirected_hdrs[key.capitalize()] = val
  252.  
  253.     
  254.     def has_header(self, header_name):
  255.         if not header_name in self.headers:
  256.             pass
  257.         return header_name in self.unredirected_hdrs
  258.  
  259.     
  260.     def get_header(self, header_name, default = None):
  261.         return self.headers.get(header_name, self.unredirected_hdrs.get(header_name, default))
  262.  
  263.     
  264.     def header_items(self):
  265.         hdrs = self.unredirected_hdrs.copy()
  266.         hdrs.update(self.headers)
  267.         return hdrs.items()
  268.  
  269.  
  270.  
  271. class OpenerDirector:
  272.     
  273.     def __init__(self):
  274.         server_version = 'Python-urllib/%s' % __version__
  275.         self.addheaders = [
  276.             ('User-agent', server_version)]
  277.         self.handlers = []
  278.         self.handle_open = { }
  279.         self.handle_error = { }
  280.         self.process_response = { }
  281.         self.process_request = { }
  282.  
  283.     
  284.     def add_handler(self, handler):
  285.         added = False
  286.         for meth in dir(handler):
  287.             i = meth.find('_')
  288.             protocol = meth[:i]
  289.             condition = meth[i + 1:]
  290.             if condition.startswith('error'):
  291.                 j = condition.find('_') + i + 1
  292.                 kind = meth[j + 1:]
  293.                 
  294.                 try:
  295.                     kind = int(kind)
  296.                 except ValueError:
  297.                     pass
  298.  
  299.                 lookup = self.handle_error.get(protocol, { })
  300.                 self.handle_error[protocol] = lookup
  301.             elif condition == 'open':
  302.                 kind = protocol
  303.                 lookup = getattr(self, 'handle_' + condition)
  304.             elif condition in [
  305.                 'response',
  306.                 'request']:
  307.                 kind = protocol
  308.                 lookup = getattr(self, 'process_' + condition)
  309.             
  310.             handlers = lookup.setdefault(kind, [])
  311.             if handlers:
  312.                 bisect.insort(handlers, handler)
  313.             else:
  314.                 handlers.append(handler)
  315.             added = True
  316.         
  317.         if added:
  318.             bisect.insort(self.handlers, handler)
  319.             handler.add_parent(self)
  320.         
  321.  
  322.     
  323.     def close(self):
  324.         pass
  325.  
  326.     
  327.     def _call_chain(self, chain, kind, meth_name, *args):
  328.         handlers = chain.get(kind, ())
  329.         for handler in handlers:
  330.             func = getattr(handler, meth_name)
  331.             result = func(*args)
  332.             if result is not None:
  333.                 return result
  334.                 continue
  335.         
  336.  
  337.     
  338.     def open(self, fullurl, data = None):
  339.         if isinstance(fullurl, basestring):
  340.             req = Request(fullurl, data)
  341.         else:
  342.             req = fullurl
  343.             if data is not None:
  344.                 req.add_data(data)
  345.             
  346.         protocol = req.get_type()
  347.         meth_name = protocol + '_request'
  348.         for processor in self.process_request.get(protocol, []):
  349.             meth = getattr(processor, meth_name)
  350.             req = meth(req)
  351.         
  352.         response = self._open(req, data)
  353.         meth_name = protocol + '_response'
  354.         for processor in self.process_response.get(protocol, []):
  355.             meth = getattr(processor, meth_name)
  356.             response = meth(req, response)
  357.         
  358.         return response
  359.  
  360.     
  361.     def _open(self, req, data = None):
  362.         result = self._call_chain(self.handle_open, 'default', 'default_open', req)
  363.         if result:
  364.             return result
  365.         
  366.         protocol = req.get_type()
  367.         result = self._call_chain(self.handle_open, protocol, protocol + '_open', req)
  368.         if result:
  369.             return result
  370.         
  371.         return self._call_chain(self.handle_open, 'unknown', 'unknown_open', req)
  372.  
  373.     
  374.     def error(self, proto, *args):
  375.         if proto in [
  376.             'http',
  377.             'https']:
  378.             dict = self.handle_error['http']
  379.             proto = args[2]
  380.             meth_name = 'http_error_%s' % proto
  381.             http_err = 1
  382.             orig_args = args
  383.         else:
  384.             dict = self.handle_error
  385.             meth_name = proto + '_error'
  386.             http_err = 0
  387.         args = (dict, proto, meth_name) + args
  388.         result = self._call_chain(*args)
  389.         if result:
  390.             return result
  391.         
  392.         if http_err:
  393.             args = (dict, 'default', 'http_error_default') + orig_args
  394.             return self._call_chain(*args)
  395.         
  396.  
  397.  
  398.  
  399. def build_opener(*handlers):
  400.     '''Create an opener object from a list of handlers.
  401.  
  402.     The opener will use several default handlers, including support
  403.     for HTTP and FTP.
  404.  
  405.     If any of the handlers passed as arguments are subclasses of the
  406.     default handlers, the default handlers will not be used.
  407.     '''
  408.     opener = OpenerDirector()
  409.     default_classes = [
  410.         ProxyHandler,
  411.         UnknownHandler,
  412.         HTTPHandler,
  413.         HTTPDefaultErrorHandler,
  414.         HTTPRedirectHandler,
  415.         FTPHandler,
  416.         FileHandler,
  417.         HTTPErrorProcessor]
  418.     if hasattr(httplib, 'HTTPS'):
  419.         default_classes.append(HTTPSHandler)
  420.     
  421.     skip = []
  422.     for klass in default_classes:
  423.         for check in handlers:
  424.             if inspect.isclass(check):
  425.                 if issubclass(check, klass):
  426.                     skip.append(klass)
  427.                 
  428.             issubclass(check, klass)
  429.             if isinstance(check, klass):
  430.                 skip.append(klass)
  431.                 continue
  432.         
  433.     
  434.     for klass in skip:
  435.         default_classes.remove(klass)
  436.     
  437.     for klass in default_classes:
  438.         opener.add_handler(klass())
  439.     
  440.     for h in handlers:
  441.         if inspect.isclass(h):
  442.             h = h()
  443.         
  444.         opener.add_handler(h)
  445.     
  446.     return opener
  447.  
  448.  
  449. class BaseHandler:
  450.     handler_order = 500
  451.     
  452.     def add_parent(self, parent):
  453.         self.parent = parent
  454.  
  455.     
  456.     def close(self):
  457.         pass
  458.  
  459.     
  460.     def __lt__(self, other):
  461.         if not hasattr(other, 'handler_order'):
  462.             return True
  463.         
  464.         return self.handler_order < other.handler_order
  465.  
  466.  
  467.  
  468. class HTTPErrorProcessor(BaseHandler):
  469.     '''Process HTTP error responses.'''
  470.     handler_order = 1000
  471.     
  472.     def http_response(self, request, response):
  473.         code = response.code
  474.         msg = response.msg
  475.         hdrs = response.info()
  476.         if code not in (200, 206):
  477.             response = self.parent.error('http', request, response, code, msg, hdrs)
  478.         
  479.         return response
  480.  
  481.     https_response = http_response
  482.  
  483.  
  484. class HTTPDefaultErrorHandler(BaseHandler):
  485.     
  486.     def http_error_default(self, req, fp, code, msg, hdrs):
  487.         raise HTTPError(req.get_full_url(), code, msg, hdrs, fp)
  488.  
  489.  
  490.  
  491. class HTTPRedirectHandler(BaseHandler):
  492.     max_repeats = 4
  493.     max_redirections = 10
  494.     
  495.     def redirect_request(self, req, fp, code, msg, headers, newurl):
  496.         """Return a Request or None in response to a redirect.
  497.  
  498.         This is called by the http_error_30x methods when a
  499.         redirection response is received.  If a redirection should
  500.         take place, return a new Request to allow http_error_30x to
  501.         perform the redirect.  Otherwise, raise HTTPError if no-one
  502.         else should try to handle this url.  Return None if you can't
  503.         but another Handler might.
  504.         """
  505.         m = req.get_method()
  506.         if (code in (301, 302, 303, 307) or m in ('GET', 'HEAD') or code in (301, 302, 303)) and m == 'POST':
  507.             return Request(newurl, headers = req.headers, origin_req_host = req.get_origin_req_host(), unverifiable = True)
  508.         else:
  509.             raise HTTPError(req.get_full_url(), code, msg, headers, fp)
  510.  
  511.     
  512.     def http_error_302(self, req, fp, code, msg, headers):
  513.         if 'location' in headers:
  514.             newurl = headers.getheaders('location')[0]
  515.         elif 'uri' in headers:
  516.             newurl = headers.getheaders('uri')[0]
  517.         else:
  518.             return None
  519.         newurl = urlparse.urljoin(req.get_full_url(), newurl)
  520.         new = self.redirect_request(req, fp, code, msg, headers, newurl)
  521.         if new is None:
  522.             return None
  523.         
  524.         if hasattr(req, 'redirect_dict'):
  525.             visited = new.redirect_dict = req.redirect_dict
  526.             if visited.get(newurl, 0) >= self.max_repeats or len(visited) >= self.max_redirections:
  527.                 raise HTTPError(req.get_full_url(), code, self.inf_msg + msg, headers, fp)
  528.             
  529.         else:
  530.             visited = new.redirect_dict = req.redirect_dict = { }
  531.         visited[newurl] = visited.get(newurl, 0) + 1
  532.         fp.read()
  533.         fp.close()
  534.         return self.parent.open(new)
  535.  
  536.     http_error_301 = http_error_303 = http_error_307 = http_error_302
  537.     inf_msg = 'The HTTP server returned a redirect error that would lead to an infinite loop.\nThe last 30x error message was:\n'
  538.  
  539.  
  540. class ProxyHandler(BaseHandler):
  541.     handler_order = 100
  542.     
  543.     def __init__(self, proxies = None):
  544.         if proxies is None:
  545.             proxies = getproxies()
  546.         
  547.         self.proxies = proxies
  548.         for type, url in proxies.items():
  549.             setattr(self, '%s_open' % type, (lambda r, proxy = url, type = type, meth = self.proxy_open: meth(r, proxy, type)))
  550.         
  551.  
  552.     
  553.     def proxy_open(self, req, proxy, type):
  554.         orig_type = req.get_type()
  555.         (type, r_type) = splittype(proxy)
  556.         (host, XXX) = splithost(r_type)
  557.         if '@' in host:
  558.             (user_pass, host) = host.split('@', 1)
  559.             if ':' in user_pass:
  560.                 (user, password) = user_pass.split(':', 1)
  561.                 user_pass = base64.encodestring('%s:%s' % (unquote(user), unquote(password))).strip()
  562.                 req.add_header('Proxy-authorization', 'Basic ' + user_pass)
  563.             
  564.         
  565.         host = unquote(host)
  566.         req.set_proxy(host, type)
  567.         if orig_type == type:
  568.             return None
  569.         else:
  570.             return self.parent.open(req)
  571.  
  572.  
  573.  
  574. class CustomProxy:
  575.     
  576.     def __init__(self, proto, func = None, proxy_addr = None):
  577.         self.proto = proto
  578.         self.func = func
  579.         self.addr = proxy_addr
  580.  
  581.     
  582.     def handle(self, req):
  583.         if self.func and self.func(req):
  584.             return 1
  585.         
  586.  
  587.     
  588.     def get_proxy(self):
  589.         return self.addr
  590.  
  591.  
  592.  
  593. class CustomProxyHandler(BaseHandler):
  594.     handler_order = 100
  595.     
  596.     def __init__(self, *proxies):
  597.         self.proxies = { }
  598.  
  599.     
  600.     def proxy_open(self, req):
  601.         proto = req.get_type()
  602.         
  603.         try:
  604.             proxies = self.proxies[proto]
  605.         except KeyError:
  606.             return None
  607.  
  608.         for p in proxies:
  609.             if p.handle(req):
  610.                 req.set_proxy(p.get_proxy())
  611.                 return self.parent.open(req)
  612.                 continue
  613.         
  614.  
  615.     
  616.     def do_proxy(self, p, req):
  617.         return self.parent.open(req)
  618.  
  619.     
  620.     def add_proxy(self, cpo):
  621.         if cpo.proto in self.proxies:
  622.             self.proxies[cpo.proto].append(cpo)
  623.         else:
  624.             self.proxies[cpo.proto] = [
  625.                 cpo]
  626.  
  627.  
  628.  
  629. class HTTPPasswordMgr:
  630.     
  631.     def __init__(self):
  632.         self.passwd = { }
  633.  
  634.     
  635.     def add_password(self, realm, uri, user, passwd):
  636.         if isinstance(uri, basestring):
  637.             uri = [
  638.                 uri]
  639.         
  640.         uri = tuple(map(self.reduce_uri, uri))
  641.         if realm not in self.passwd:
  642.             self.passwd[realm] = { }
  643.         
  644.         self.passwd[realm][uri] = (user, passwd)
  645.  
  646.     
  647.     def find_user_password(self, realm, authuri):
  648.         domains = self.passwd.get(realm, { })
  649.         authuri = self.reduce_uri(authuri)
  650.         for uris, authinfo in domains.iteritems():
  651.             for uri in uris:
  652.                 if self.is_suburi(uri, authuri):
  653.                     return authinfo
  654.                     continue
  655.             
  656.         
  657.         return (None, None)
  658.  
  659.     
  660.     def reduce_uri(self, uri):
  661.         '''Accept netloc or URI and extract only the netloc and path'''
  662.         parts = urlparse.urlparse(uri)
  663.         if parts[1]:
  664.             if not parts[2]:
  665.                 pass
  666.             return (parts[1], '/')
  667.         else:
  668.             return (parts[2], '/')
  669.  
  670.     
  671.     def is_suburi(self, base, test):
  672.         '''Check if test is below base in a URI tree
  673.  
  674.         Both args must be URIs in reduced form.
  675.         '''
  676.         if base == test:
  677.             return True
  678.         
  679.         if base[0] != test[0]:
  680.             return False
  681.         
  682.         common = posixpath.commonprefix((base[1], test[1]))
  683.         if len(common) == len(base[1]):
  684.             return True
  685.         
  686.         return False
  687.  
  688.  
  689.  
  690. class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
  691.     
  692.     def find_user_password(self, realm, authuri):
  693.         (user, password) = HTTPPasswordMgr.find_user_password(self, realm, authuri)
  694.         if user is not None:
  695.             return (user, password)
  696.         
  697.         return HTTPPasswordMgr.find_user_password(self, None, authuri)
  698.  
  699.  
  700.  
  701. class AbstractBasicAuthHandler:
  702.     rx = re.compile('[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', re.I)
  703.     
  704.     def __init__(self, password_mgr = None):
  705.         if password_mgr is None:
  706.             password_mgr = HTTPPasswordMgr()
  707.         
  708.         self.passwd = password_mgr
  709.         self.add_password = self.passwd.add_password
  710.  
  711.     
  712.     def http_error_auth_reqed(self, authreq, host, req, headers):
  713.         authreq = headers.get(authreq, None)
  714.         if authreq:
  715.             mo = AbstractBasicAuthHandler.rx.search(authreq)
  716.             if mo:
  717.                 (scheme, realm) = mo.groups()
  718.                 if scheme.lower() == 'basic':
  719.                     return self.retry_http_basic_auth(host, req, realm)
  720.                 
  721.             
  722.         
  723.  
  724.     
  725.     def retry_http_basic_auth(self, host, req, realm):
  726.         (user, pw) = self.passwd.find_user_password(realm, host)
  727.         if pw is not None:
  728.             raw = '%s:%s' % (user, pw)
  729.             auth = 'Basic %s' % base64.encodestring(raw).strip()
  730.             if req.headers.get(self.auth_header, None) == auth:
  731.                 return None
  732.             
  733.             req.add_header(self.auth_header, auth)
  734.             return self.parent.open(req)
  735.         else:
  736.             return None
  737.  
  738.  
  739.  
  740. class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
  741.     auth_header = 'Authorization'
  742.     
  743.     def http_error_401(self, req, fp, code, msg, headers):
  744.         host = urlparse.urlparse(req.get_full_url())[1]
  745.         return self.http_error_auth_reqed('www-authenticate', host, req, headers)
  746.  
  747.  
  748.  
  749. class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
  750.     auth_header = 'Proxy-authorization'
  751.     
  752.     def http_error_407(self, req, fp, code, msg, headers):
  753.         host = req.get_host()
  754.         return self.http_error_auth_reqed('proxy-authenticate', host, req, headers)
  755.  
  756.  
  757.  
  758. def randombytes(n):
  759.     '''Return n random bytes.'''
  760.     pass
  761.  
  762.  
  763. class AbstractDigestAuthHandler:
  764.     
  765.     def __init__(self, passwd = None):
  766.         if passwd is None:
  767.             passwd = HTTPPasswordMgr()
  768.         
  769.         self.passwd = passwd
  770.         self.add_password = self.passwd.add_password
  771.         self.retried = 0
  772.         self.nonce_count = 0
  773.  
  774.     
  775.     def reset_retry_count(self):
  776.         self.retried = 0
  777.  
  778.     
  779.     def http_error_auth_reqed(self, auth_header, host, req, headers):
  780.         authreq = headers.get(auth_header, None)
  781.         if authreq:
  782.             scheme = authreq.split()[0]
  783.             if scheme.lower() == 'digest':
  784.                 return self.retry_http_digest_auth(req, authreq)
  785.             else:
  786.                 raise ValueError("AbstractDigestAuthHandler doesn't know about %s" % scheme)
  787.         
  788.  
  789.     
  790.     def retry_http_digest_auth(self, req, auth):
  791.         (token, challenge) = auth.split(' ', 1)
  792.         chal = parse_keqv_list(parse_http_list(challenge))
  793.         auth = self.get_authorization(req, chal)
  794.         if auth:
  795.             auth_val = 'Digest %s' % auth
  796.             if req.headers.get(self.auth_header, None) == auth_val:
  797.                 return None
  798.             
  799.             req.add_header(self.auth_header, auth_val)
  800.             resp = self.parent.open(req)
  801.             return resp
  802.         
  803.  
  804.     
  805.     def get_cnonce(self, nonce):
  806.         dig = sha.new('%s:%s:%s:%s' % (self.nonce_count, nonce, time.ctime(), randombytes(8))).hexdigest()
  807.         return dig[:16]
  808.  
  809.     
  810.     def get_authorization(self, req, chal):
  811.         
  812.         try:
  813.             realm = chal['realm']
  814.             nonce = chal['nonce']
  815.             qop = chal.get('qop')
  816.             algorithm = chal.get('algorithm', 'MD5')
  817.             opaque = chal.get('opaque', None)
  818.         except KeyError:
  819.             return None
  820.  
  821.         (H, KD) = self.get_algorithm_impls(algorithm)
  822.         if H is None:
  823.             return None
  824.         
  825.         (user, pw) = self.passwd.find_user_password(realm, req.get_full_url())
  826.         if user is None:
  827.             return None
  828.         
  829.         if req.has_data():
  830.             entdig = self.get_entity_digest(req.get_data(), chal)
  831.         else:
  832.             entdig = None
  833.         A1 = '%s:%s:%s' % (user, realm, pw)
  834.         A2 = '%s:%s' % (req.get_method(), req.get_selector())
  835.         if qop == 'auth':
  836.             self.nonce_count += 1
  837.             ncvalue = '%08x' % self.nonce_count
  838.             cnonce = self.get_cnonce(nonce)
  839.             noncebit = '%s:%s:%s:%s:%s' % (nonce, ncvalue, cnonce, qop, H(A2))
  840.             respdig = KD(H(A1), noncebit)
  841.         elif qop is None:
  842.             respdig = KD(H(A1), '%s:%s' % (nonce, H(A2)))
  843.         
  844.         base = 'username="%s", realm="%s", nonce="%s", uri="%s", response="%s"' % (user, realm, nonce, req.get_selector(), respdig)
  845.         if opaque:
  846.             base = base + ', opaque="%s"' % opaque
  847.         
  848.         if entdig:
  849.             base = base + ', digest="%s"' % entdig
  850.         
  851.         if algorithm != 'MD5':
  852.             base = base + ', algorithm="%s"' % algorithm
  853.         
  854.         if qop:
  855.             base = base + ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
  856.         
  857.         return base
  858.  
  859.     
  860.     def get_algorithm_impls(self, algorithm):
  861.         if algorithm == 'MD5':
  862.             
  863.             H = lambda x: md5.new(x).hexdigest()
  864.         elif algorithm == 'SHA':
  865.             
  866.             H = lambda x: sha.new(x).hexdigest()
  867.         
  868.         
  869.         KD = lambda s, d: H('%s:%s' % (s, d))
  870.         return (H, KD)
  871.  
  872.     
  873.     def get_entity_digest(self, data, chal):
  874.         pass
  875.  
  876.  
  877.  
  878. class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
  879.     '''An authentication protocol defined by RFC 2069
  880.  
  881.     Digest authentication improves on basic authentication because it
  882.     does not transmit passwords in the clear.
  883.     '''
  884.     auth_header = 'Authorization'
  885.     
  886.     def http_error_401(self, req, fp, code, msg, headers):
  887.         host = urlparse.urlparse(req.get_full_url())[1]
  888.         retry = self.http_error_auth_reqed('www-authenticate', host, req, headers)
  889.         self.reset_retry_count()
  890.         return retry
  891.  
  892.  
  893.  
  894. class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
  895.     auth_header = 'Proxy-Authorization'
  896.     
  897.     def http_error_407(self, req, fp, code, msg, headers):
  898.         host = req.get_host()
  899.         retry = self.http_error_auth_reqed('proxy-authenticate', host, req, headers)
  900.         self.reset_retry_count()
  901.         return retry
  902.  
  903.  
  904.  
  905. class AbstractHTTPHandler(BaseHandler):
  906.     
  907.     def __init__(self, debuglevel = 0):
  908.         self._debuglevel = debuglevel
  909.  
  910.     
  911.     def set_http_debuglevel(self, level):
  912.         self._debuglevel = level
  913.  
  914.     
  915.     def do_request_(self, request):
  916.         host = request.get_host()
  917.         if not host:
  918.             raise URLError('no host given')
  919.         
  920.         if request.has_data():
  921.             data = request.get_data()
  922.             if not request.has_header('Content-type'):
  923.                 request.add_unredirected_header('Content-type', 'application/x-www-form-urlencoded')
  924.             
  925.             if not request.has_header('Content-length'):
  926.                 request.add_unredirected_header('Content-length', '%d' % len(data))
  927.             
  928.         
  929.         (scheme, sel) = splittype(request.get_selector())
  930.         (sel_host, sel_path) = splithost(sel)
  931.         if not request.has_header('Host'):
  932.             if not sel_host:
  933.                 pass
  934.             request.add_unredirected_header('Host', host)
  935.         
  936.         for name, value in self.parent.addheaders:
  937.             name = name.capitalize()
  938.             if not request.has_header(name):
  939.                 request.add_unredirected_header(name, value)
  940.                 continue
  941.         
  942.         return request
  943.  
  944.     
  945.     def do_open(self, http_class, req):
  946.         '''Return an addinfourl object for the request, using http_class.
  947.  
  948.         http_class must implement the HTTPConnection API from httplib.
  949.         The addinfourl return value is a file-like object.  It also
  950.         has methods and attributes including:
  951.             - info(): return a mimetools.Message object for the headers
  952.             - geturl(): return the original request URL
  953.             - code: HTTP status code
  954.         '''
  955.         host = req.get_host()
  956.         if not host:
  957.             raise URLError('no host given')
  958.         
  959.         h = http_class(host)
  960.         h.set_debuglevel(self._debuglevel)
  961.         headers = dict(req.headers)
  962.         headers.update(req.unredirected_hdrs)
  963.         headers['Connection'] = 'close'
  964.         
  965.         try:
  966.             h.request(req.get_method(), req.get_selector(), req.data, headers)
  967.             r = h.getresponse()
  968.         except socket.error:
  969.             err = None
  970.             raise URLError(err)
  971.  
  972.         r.recv = r.read
  973.         fp = socket._fileobject(r)
  974.         resp = addinfourl(fp, r.msg, req.get_full_url())
  975.         resp.code = r.status
  976.         resp.msg = r.reason
  977.         return resp
  978.  
  979.  
  980.  
  981. class HTTPHandler(AbstractHTTPHandler):
  982.     
  983.     def http_open(self, req):
  984.         return self.do_open(httplib.HTTPConnection, req)
  985.  
  986.     http_request = AbstractHTTPHandler.do_request_
  987.  
  988. if hasattr(httplib, 'HTTPS'):
  989.     
  990.     class HTTPSHandler(AbstractHTTPHandler):
  991.         
  992.         def https_open(self, req):
  993.             return self.do_open(httplib.HTTPSConnection, req)
  994.  
  995.         https_request = AbstractHTTPHandler.do_request_
  996.  
  997.  
  998.  
  999. class HTTPCookieProcessor(BaseHandler):
  1000.     
  1001.     def __init__(self, cookiejar = None):
  1002.         if cookiejar is None:
  1003.             cookiejar = cookielib.CookieJar()
  1004.         
  1005.         self.cookiejar = cookiejar
  1006.  
  1007.     
  1008.     def http_request(self, request):
  1009.         self.cookiejar.add_cookie_header(request)
  1010.         return request
  1011.  
  1012.     
  1013.     def http_response(self, request, response):
  1014.         self.cookiejar.extract_cookies(response, request)
  1015.         return response
  1016.  
  1017.     https_request = http_request
  1018.     https_response = http_response
  1019.  
  1020.  
  1021. class UnknownHandler(BaseHandler):
  1022.     
  1023.     def unknown_open(self, req):
  1024.         type = req.get_type()
  1025.         raise URLError('unknown url type: %s' % type)
  1026.  
  1027.  
  1028.  
  1029. def parse_keqv_list(l):
  1030.     '''Parse list of key=value strings where keys are not duplicated.'''
  1031.     parsed = { }
  1032.     for elt in l:
  1033.         (k, v) = elt.split('=', 1)
  1034.         if v[0] == '"' and v[-1] == '"':
  1035.             v = v[1:-1]
  1036.         
  1037.         parsed[k] = v
  1038.     
  1039.     return parsed
  1040.  
  1041.  
  1042. def parse_http_list(s):
  1043.     '''Parse lists as described by RFC 2068 Section 2.
  1044.     
  1045.     In particular, parse comma-separated lists where the elements of
  1046.     the list may include quoted-strings.  A quoted-string could
  1047.     contain a comma.  A non-quoted string could have quotes in the
  1048.     middle.  Neither commas nor quotes count if they are escaped.
  1049.     Only double-quotes count, not single-quotes.
  1050.     '''
  1051.     res = []
  1052.     part = ''
  1053.     escape = quote = False
  1054.     for cur in s:
  1055.         if escape:
  1056.             part += cur
  1057.             escape = False
  1058.             continue
  1059.         
  1060.         if quote:
  1061.             if cur == '\\':
  1062.                 escape = True
  1063.                 continue
  1064.             elif cur == '"':
  1065.                 quote = False
  1066.             
  1067.             part += cur
  1068.             continue
  1069.         
  1070.         if cur == ',':
  1071.             res.append(part)
  1072.             part = ''
  1073.             continue
  1074.         
  1075.         if cur == '"':
  1076.             quote = True
  1077.         
  1078.         part += cur
  1079.     
  1080.     if part:
  1081.         res.append(part)
  1082.     
  1083.     return [ part.strip() for part in res ]
  1084.  
  1085.  
  1086. class FileHandler(BaseHandler):
  1087.     
  1088.     def file_open(self, req):
  1089.         url = req.get_selector()
  1090.         if url[:2] == '//' and url[2:3] != '/':
  1091.             req.type = 'ftp'
  1092.             return self.parent.open(req)
  1093.         else:
  1094.             return self.open_local_file(req)
  1095.  
  1096.     names = None
  1097.     
  1098.     def get_names(self):
  1099.         if FileHandler.names is None:
  1100.             FileHandler.names = (socket.gethostbyname('localhost'), socket.gethostbyname(socket.gethostname()))
  1101.         
  1102.         return FileHandler.names
  1103.  
  1104.     
  1105.     def open_local_file(self, req):
  1106.         import email.Utils as email
  1107.         host = req.get_host()
  1108.         file = req.get_selector()
  1109.         localfile = url2pathname(file)
  1110.         stats = os.stat(localfile)
  1111.         size = stats.st_size
  1112.         modified = email.Utils.formatdate(stats.st_mtime, usegmt = True)
  1113.         mtype = mimetypes.guess_type(file)[0]
  1114.         if not mtype:
  1115.             pass
  1116.         headers = mimetools.Message(StringIO('Content-type: %s\nContent-length: %d\nLast-modified: %s\n' % ('text/plain', size, modified)))
  1117.         if host:
  1118.             (host, port) = splitport(host)
  1119.         
  1120.         if (not host or not port) and socket.gethostbyname(host) in self.get_names():
  1121.             return addinfourl(open(localfile, 'rb'), headers, 'file:' + file)
  1122.         
  1123.         raise URLError('file not on local host')
  1124.  
  1125.  
  1126.  
  1127. class FTPHandler(BaseHandler):
  1128.     
  1129.     def ftp_open(self, req):
  1130.         host = req.get_host()
  1131.         if not host:
  1132.             raise IOError, ('ftp error', 'no host given')
  1133.         
  1134.         (host, port) = splitport(host)
  1135.         if port is None:
  1136.             port = ftplib.FTP_PORT
  1137.         else:
  1138.             port = int(port)
  1139.         (user, host) = splituser(host)
  1140.         if user:
  1141.             (user, passwd) = splitpasswd(user)
  1142.         else:
  1143.             passwd = None
  1144.         host = unquote(host)
  1145.         if not user:
  1146.             pass
  1147.         user = unquote('')
  1148.         if not passwd:
  1149.             pass
  1150.         passwd = unquote('')
  1151.         
  1152.         try:
  1153.             host = socket.gethostbyname(host)
  1154.         except socket.error:
  1155.             msg = None
  1156.             raise URLError(msg)
  1157.  
  1158.         (path, attrs) = splitattr(req.get_selector())
  1159.         dirs = path.split('/')
  1160.         dirs = map(unquote, dirs)
  1161.         dirs = dirs[:-1]
  1162.         file = dirs[-1]
  1163.         if dirs and not dirs[0]:
  1164.             dirs = dirs[1:]
  1165.         
  1166.         
  1167.         try:
  1168.             fw = self.connect_ftp(user, passwd, host, port, dirs)
  1169.             if not file or 'I':
  1170.                 pass
  1171.             type = 'D'
  1172.             for attr in attrs:
  1173.                 (attr, value) = splitvalue(attr)
  1174.                 if attr.lower() == 'type' and value in ('a', 'A', 'i', 'I', 'd', 'D'):
  1175.                     type = value.upper()
  1176.                     continue
  1177.             
  1178.             (fp, retrlen) = fw.retrfile(file, type)
  1179.             headers = ''
  1180.             mtype = mimetypes.guess_type(req.get_full_url())[0]
  1181.             if mtype:
  1182.                 headers += 'Content-type: %s\n' % mtype
  1183.             
  1184.             if retrlen is not None and retrlen >= 0:
  1185.                 headers += 'Content-length: %d\n' % retrlen
  1186.             
  1187.             sf = StringIO(headers)
  1188.             headers = mimetools.Message(sf)
  1189.             return addinfourl(fp, headers, req.get_full_url())
  1190.         except ftplib.all_errors:
  1191.             msg = None
  1192.             raise IOError, ('ftp error', msg), sys.exc_info()[2]
  1193.  
  1194.  
  1195.     
  1196.     def connect_ftp(self, user, passwd, host, port, dirs):
  1197.         fw = ftpwrapper(user, passwd, host, port, dirs)
  1198.         return fw
  1199.  
  1200.  
  1201.  
  1202. class CacheFTPHandler(FTPHandler):
  1203.     
  1204.     def __init__(self):
  1205.         self.cache = { }
  1206.         self.timeout = { }
  1207.         self.soonest = 0
  1208.         self.delay = 60
  1209.         self.max_conns = 16
  1210.  
  1211.     
  1212.     def setTimeout(self, t):
  1213.         self.delay = t
  1214.  
  1215.     
  1216.     def setMaxConns(self, m):
  1217.         self.max_conns = m
  1218.  
  1219.     
  1220.     def connect_ftp(self, user, passwd, host, port, dirs):
  1221.         key = (user, host, port, '/'.join(dirs))
  1222.         if key in self.cache:
  1223.             self.timeout[key] = time.time() + self.delay
  1224.         else:
  1225.             self.cache[key] = ftpwrapper(user, passwd, host, port, dirs)
  1226.             self.timeout[key] = time.time() + self.delay
  1227.         self.check_cache()
  1228.         return self.cache[key]
  1229.  
  1230.     
  1231.     def check_cache(self):
  1232.         t = time.time()
  1233.         if self.soonest <= t:
  1234.             for k, v in self.timeout.items():
  1235.                 if v < t:
  1236.                     self.cache[k].close()
  1237.                     del self.cache[k]
  1238.                     del self.timeout[k]
  1239.                     continue
  1240.             
  1241.         
  1242.         self.soonest = min(self.timeout.values())
  1243.         if len(self.cache) == self.max_conns:
  1244.             for k, v in self.timeout.items():
  1245.                 if v == self.soonest:
  1246.                     del self.cache[k]
  1247.                     del self.timeout[k]
  1248.                     break
  1249.                     continue
  1250.             
  1251.             self.soonest = min(self.timeout.values())
  1252.         
  1253.  
  1254.  
  1255.  
  1256. class GopherHandler(BaseHandler):
  1257.     
  1258.     def gopher_open(self, req):
  1259.         host = req.get_host()
  1260.         if not host:
  1261.             raise GopherError('no host given')
  1262.         
  1263.         host = unquote(host)
  1264.         selector = req.get_selector()
  1265.         (type, selector) = splitgophertype(selector)
  1266.         (selector, query) = splitquery(selector)
  1267.         selector = unquote(selector)
  1268.         if query:
  1269.             query = unquote(query)
  1270.             fp = gopherlib.send_query(selector, query, host)
  1271.         else:
  1272.             fp = gopherlib.send_selector(selector, host)
  1273.         return addinfourl(fp, noheaders(), req.get_full_url())
  1274.  
  1275.  
  1276.  
  1277. class OpenerFactory:
  1278.     default_handlers = [
  1279.         UnknownHandler,
  1280.         HTTPHandler,
  1281.         HTTPDefaultErrorHandler,
  1282.         HTTPRedirectHandler,
  1283.         FTPHandler,
  1284.         FileHandler]
  1285.     handlers = []
  1286.     replacement_handlers = []
  1287.     
  1288.     def add_handler(self, h):
  1289.         self.handlers = self.handlers + [
  1290.             h]
  1291.  
  1292.     
  1293.     def replace_handler(self, h):
  1294.         pass
  1295.  
  1296.     
  1297.     def build_opener(self):
  1298.         opener = OpenerDirector()
  1299.         for ph in self.default_handlers:
  1300.             if inspect.isclass(ph):
  1301.                 ph = ph()
  1302.             
  1303.             opener.add_handler(ph)
  1304.         
  1305.  
  1306.  
  1307.